package com.algorithm.ch1.lzr.linkedlist;

import java.util.ArrayList;
import java.util.List;

/**
 * 删除链表的倒数第N个节点
 *
 * @author lzr
 * @date 2018/8/16
 */
public class RemoveNthFromEnd {

    public static void main(String[] args) {
        ListNode listNode1 = new ListNode(1);
        ListNode listNode2 = new ListNode(2);
        ListNode listNode3 = new ListNode(3);
        ListNode listNode4 = new ListNode(4);
        ListNode listNode5 = new ListNode(5);
        listNode1.next = listNode2;
        //listNode2.next = listNode3;
        //listNode3.next = listNode4;
        //listNode4.next = listNode5;
        ListNode listNode = removeNthFromEnd(listNode1, 1);
        System.out.print(listNode.val);
    }

    /**
     * 方式1: 利用List集合(尝试直接操作链表直接解决)
     * 复杂度T(n) = N
     *
     * @param head
     * @param n
     * @return
     */
    public static ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode nextNode = head.next;
        if (nextNode == null) {
            return null;
        }
        List<ListNode> list = new ArrayList<>();
        int i = 1;
        list.add(head);
        while (nextNode != null) {
            list.add(nextNode);
            nextNode = nextNode.next;
            i++;
        }
        if (n == 1) {
            ListNode destNode = list.get(i - n - 1);
            destNode.next = null;
        } else {
            ListNode destNode = list.get(i - n);
            destNode.val = destNode.next.val;
            destNode.next = destNode.next.next;
        }
        return head;
    }
}
